/**
 * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef __INC_LLT_GE_EXECUTOR_STUB_H
#define __INC_LLT_GE_EXECUTOR_STUB_H
#include <vector>
#include <gtest/gtest.h>
#include <gmock/gmock.h>
#include "acl/acl_base.h"
#include "framework/executor_c/ge_executor.h"
#include "runtime/rt.h"

class GeExecutorStubMock {
public:
  static GeExecutorStubMock& GetInstance() {
    static GeExecutorStubMock mock;
    return mock;
  }

  MOCK_METHOD2(GetPartitionSize, Status(const char *fileName, GePartitionSize *mdlPartitionSize));
  MOCK_METHOD3(GetMemAndWeightSize, Status(const char *fileName, size_t *workSize, size_t *weightSize));
  MOCK_METHOD5(ExecModel, Status(uint32_t modelId, ExecHandleDesc *execDesc, bool sync,
               const InputData *inputData, OutputData *outputData));
  MOCK_METHOD2(GeLoadModelFromData, Status(uint32_t *modelId, const ModelData *modelData));
  MOCK_METHOD1(UnloadModel, Status(uint32_t modelId));
  MOCK_METHOD1(DestoryModelInOutInfo, Status(ModelInOutInfo *info));
  MOCK_METHOD0(GeInitialize, Status());
  MOCK_METHOD0(GeFinalize, Status());
  MOCK_METHOD1(GeDbgInit, Status(const char *configPath));
  MOCK_METHOD0(GeDbgDeInit, Status());
  MOCK_METHOD2(GeNofifySetDevice, Status(uint32_t chipId, uint32_t deviceId));
};

#endif